import matplotlib.pyplot as plt  
X1 = range(0, 50)  
Y1 = [num**2 for num in X1] # y = x^2 X2 = [0, 1] Y2 = [0, 1] # y = x  
  
Fig = plt.figure(figsize=(8,4)) # Create a `figure' instance  
Ax = Fig.add_subplot(111) # Create a `axes' instance in the figure  
Ax.plot(X1, Y1) # Create a Line2D instance in the axes  
  
Fig.show()
